knn algorithm example in python